#pragma once

#include <map>

using namespace std;

class Soldier;

typedef std::map<unsigned int, Soldier*> SOLDIERCONTAINER;

class SoldierContainer
{
public:
	SoldierContainer();
	~SoldierContainer();

	void AddSoldier(Soldier* pSoldier);
	void RemoveSoldier(const Soldier* pSoldier);

	Soldier* GetSoldier(unsigned int soldierid);
	SOLDIERCONTAINER GetSoldierContainer();

private:
	bool CheckSoldier(unsigned int soldierid);

	SOLDIERCONTAINER m_mapSoldiers;
};